package 面试经典150题.二叉树;

import common.TreeNode;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * @author tmh
 * @date 2024/4/7 17:06
 * @description
 */
public class T103二叉树的锯齿形层序遍历 {
    public static void main(String[] args) {

    }

    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> result = new ArrayList<>();
        if (root == null) {
            return result;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        //层序遍历模板
        while (queue.size() != 0) {
            LinkedList<Integer> temp = new LinkedList<>();
            //这里先让i初始值为队列的长度
            for (int i = queue.size(); i > 0; i--) {
                //需要拿出一个变量来接受队列里面弹出的元素
                TreeNode node = queue.poll();
                if (result.size() % 2 == 0) {
                    //说明队列里面已经存储了偶数层，下一层就是顺序排列
                    temp.addLast(node.val);
                } else {
                    //说明队列里面已经存储了奇数层，下一层就是逆序排列
                    temp.addFirst(node.val);
                }
                if (node.left != null) {
                    queue.add(node.left);
                }
                if (node.right != null) {
                    queue.add(node.right);
                }
            }
            result.add(temp);
        }
        return result;
    }
}
